import numpy as np
import torch
from torch.utils.data import Dataset

from model.ours_LFDS.utils import get_lds_kernel_window
from scipy.ndimage import convolve1d

class MyDataset(Dataset):
    def __init__(self, X, y=None):
        self.X = X
        self.y = y

    def __len__(self):
        return len(self.X)

    def __getitem__(self, index):
        # print(self.X[index])
        # print(self.y.iloc[index])


        return self.X[index], self.y[index]

        # if self.lds:
        #     return self.X[index], self.y[index], weight
        # else:
        #     return self.X[index], self.y[index]
